from pydantic import BaseModel, Field

'''
水敏实验
'''


class GetExperimentParam(BaseModel):
    pitUnits: list = Field(title="井")
    date: str = Field(title="时间", default=None)


class AddExperimentParam(BaseModel):
    core_id: str = Field(title="岩芯")
    date: str = Field(title="时间", default=None)


class DeleteExperimentParam(BaseModel):
    delete: list = Field(title="删除列表", default=[{"id": 1}])


class waterExperimentUpdate(BaseModel):
    delete: list = Field(title="删除", default=[
        {'Core_ID': -1, "Model_ID": -1}])
    update: list = Field(title="修改",
                         default=[
                             {'Core_ID': 1}])

class DetailExperimentParam(BaseModel):
    id: int = Field(title="岩芯id")


class UpdateExperimentParam(BaseModel):
    id: int = Field(title="id")
    data: list = Field(title='实验数据', default=[{
        'inject_fluid': '#',
        'Inject_volume': 10,
        'permeability': 1,
        'damage_ratio': 1,
    }, {
        'inject_fluid': '#',
        'Inject_volume': 10,
        'permeability': 1,
        'damage_ratio': 1,
    }])
